package gsredis

import (
	"fmt"

	"github.com/go-redis/redis"
)

// 获取指定节点的 redis 实例。
//
//	@nodeName 节点名称。
func Get(nodeName string) *redis.Client {
	if RedisManager == nil {
		panic("redis service is not register!")
	}

	if v, ok := RedisManager.dbMap[nodeName]; ok {
		return v
	}

	panic(fmt.Sprintf("redis node %s does not exist", nodeName))
}

// 注册 redis 服务。
func Register(config map[string]*RedisConfig) {
	RedisManager = newRedisManager()
	RedisManager.init(config)
}
